load("//tensorflow:tensorflow.bzl", "tf_custom_op_library")

tf_custom_op_library(
    name = "shift_corr.so",
    srcs = ["shift_corr.cc"],
    gpu_srcs = ["shift_corr.cu.cc"],
)
